package com.kmxd.ams.infra.tmpl.util;

import com.kmxd.ams.infra.tmpl.dto.ArchTmplBO;
import com.kmxd.ams.infra.tmpl.enums.TmplEnum;
import com.kmxd.ams.infra.tmpl.tmpl.AbstractTmpl;
import java.util.List;
import java.util.Map;
import lombok.AllArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;

@Component
@Slf4j
@AllArgsConstructor
public class TmplUtils {

  private Map<String, AbstractTmpl> serviceMap;

  /**
   * 获取模板字段
   *
   * @return
   */
  public List<ArchTmplBO> getTmpl(String tmplEnumStr) {
    TmplEnum tmplEnum = TmplEnum.convert(tmplEnumStr);
    return serviceMap.get(tmplEnum.getCode()).getTmpl();
  }
}
